package main

import (
	"fmt"
	"golang.org/x/tour/reader"
	"io"
	"os"
	"strings"
)

/*
https://tour.go-zh.org/methods/22
练习：Reader
实现一个 Reader 类型，它产生一个 ASCII 字符 'A' 的无限流。

https://tour.go-zh.org/methods/23
练习：rot13Reader
有种常见的模式是一个 io.Reader 包装另一个 io.Reader，然后通过某种方式修改其数据流。

例如，gzip.NewReader 函数接受一个 io.Reader（已压缩的数据流）并返回一个同样实现了 io.Reader 的 *gzip.Reader（解压后的数据流）。

编写一个实现了 io.Reader 并从另一个 io.Reader 中读取数据的 rot13Reader，通过应用 rot13 代换密码对数据流进行修改。

rot13Reader 类型已经提供。实现 Read 方法以满足 io.Reader。
*/

type MyReader struct{}

func (r MyReader) Read(p []byte) (n int, err error) {
	for i := 0; i < len(p); i++ {
		p[i] = byte('A')
	}
	n = len(p)
	err = nil

	return
}

type rot13Reader struct {
	r io.Reader
}

func (reader rot13Reader) Read(b []byte) (n int, err error) {
	n, err = reader.r.Read(b)
	for i := 0; i < n; i++ {
		switch {
		case b[i] >= 'A' && b[i] < 'N', b[i] >= 'a' && b[i] < 'n':
			b[i] += 13
		case b[i] >= 'N' && b[i] <= 'Z', b[i] >= 'n' && b[i] <= 'z':
			b[i] -= 13
		}
	}
	return
}

func main() {
	reader.Validate(MyReader{})
	fmt.Println("----DemoRot13Reader-----")
	DemoRot13Reader()
}

func DemoRot13Reader() {
	s := strings.NewReader("Lbh penpxrq gur pbqr!")
	r := rot13Reader{s}
	io.Copy(os.Stdout, &r)

}
